package java学习.leetcode.editor.cn;

import java.util.ArrayList;
import java.util.Stack;

/**
 * @author 刘世锦
 * 2023-01-16 15:56:09	 当前时间
 */
//给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于
// node.val 的值之和。 
//
// 提醒一下，二叉搜索树满足下列约束条件： 
//
// 
// 节点的左子树仅包含键 小于 节点键的节点。 
// 节点的右子树仅包含键 大于 节点键的节点。 
// 左右子树也必须是二叉搜索树。 
// 
//
// 注意：本题和 1038: https://leetcode-cn.com/problems/binary-search-tree-to-greater-s
//um-tree/ 相同 
//
// 
//
// 示例 1： 
//
// 
//
// 输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
//输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]
// 
//
// 示例 2： 
//
// 输入：root = [0,null,1]
//输出：[1,null,1]
// 
//
// 示例 3： 
//
// 输入：root = [1,0,2]
//输出：[3,3,2]
// 
//
// 示例 4： 
//
// 输入：root = [3,2,4,1]
//输出：[7,9,4,10]
// 
//
// 
//
// 提示： 
//
// 
// 树中的节点数介于 0 和 104 之间。 
// 每个节点的值介于 -104 和 104 之间。 
// 树中的所有值 互不相同 。 
// 给定的树为二叉搜索树。 
// 
// Related Topics 树 深度优先搜索 二叉搜索树 二叉树 
// 👍 826 👎 0

public class 把二叉搜索树转换为累加树{
	public static void main(String[] args) {
		Solution solution = new 把二叉搜索树转换为累加树().new Solution();
		
	}
//leetcode submit region begin(Prohibit modification and deletion)
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {



	/** 复习  @author 刘世锦
	 *  @date  2023/2/11 16:51
	 *
	 *  | 4
	 *
	 *   8  7  6
	 *     15  21
	 */

	// 右中左
	public TreeNode convertBST(TreeNode root) {
		Stack<TreeNode> stack = new Stack<>();
		TreeNode result = root,pre = null;
		while (!stack.isEmpty()||root!=null){
			while (root!=null){
				stack.push(root);
				root  =root.right;
			}
			TreeNode node = stack.pop();
			if (pre!=null){
				node.val+=pre.val;
			}
			pre = node;
			root = node.left;

		}
		return result;


	}

		TreeNode pre = null;
	public TreeNode convertBSTByReview(TreeNode root) {
		if (root==null){
			return root;
		}

		TreeNode right = convertBST(root.right);

		if (pre!=null){
			root.val +=pre.val;

		}
		pre = root;
		TreeNode left = convertBST(root.left);
		root.left = left;
		root.right = right;
		return root;


	}

	// 复习end

	// 中序
	public TreeNode convertBSTDFS(TreeNode root) {
		Stack<TreeNode> stack = new Stack<>();
//		TreeNode  pre  = null;
		int sum = 0;
		TreeNode cur = root;
		if (root!=null){
			while (!stack.isEmpty()||root!=null){
				while (root!=null){
					stack.push(root);
					root = root.right;
				}
				root = stack.pop();
				// 操作 root
//				if (pre!=null){
//					root.val +=pre.val;
//				}
//				pre = root;
				sum+=root.val;
				root.val = sum;
				root = root.left;

			}
		}
		return cur;
	}

		/**
         *  1  4  6
         * 11  10 6
         *
         *   6  4  1
         *  6  10  11
         *  逆中序  右中左 时，累加树为依次递增
         *  使用num 记录前一节点的val 做累加，每次root都累加num
         *
         */
	int num;
    public TreeNode convertBST1(TreeNode root) {
		if(root==null){
			return root;
		}
		TreeNode right = convertBST(root.right);
			root.val += num;
			num = root.val;
		TreeNode left = convertBST(root.left);
//		if (left!=null){
//			left.val +=root.val;
//		}
		return root;

    }
}
//leetcode submit region end(Prohibit modification and deletion)

}
